from openai import OpenAI
class LLMModel:
    def __init__(self):
        self.client = OpenAI(
            api_key="********",  # 在这里将 MOONSHOT_API_KEY 替换为你从 Kimi 开放平台申请的 API Key
            base_url="https://api.moonshot.cn/v1",
        )
    def generate(self, question):
        """
        使用 Moonshot AI 的 Kimi 大模型生成回答
        :param question: 用户提问
        :return: Kimi 大模型的回答
        """
        # 调用 Moonshot AI 的 Kimi 大模型进行回答生成
        completion = self.client.chat.completions.create(
            model="moonshot-v1-8k",
            messages=[
                {
                    "role": "system",
                    "content": "你是 Kimi，由 Moonshot AI 提供的人工智能助手，你更擅长中文和英文的对话。你会为用户提供安全，有帮助，准确的回答。同时，你会拒绝一切涉及恐怖主义，种族歧视，黄色暴力等问题的回答。Moonshot AI 为专有名词，不可翻译成其他语言。",
                },
                {"role": "user", "content": question},
            ],

            temperature=0.3,
        )
        return completion.choices[0].message.content
